package 算法.leetcode.offer;

/**
 * 在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。
 * 请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
 *
 *  
 *
 * 示例:
 *
 * 现有矩阵 matrix 如下：
 *
 * [
 *   [1,   4,  7, 11, 15],
 *   [2,   5,  8, 12, 19],
 *   [3,   6,  9, 16, 22],
 *   [10, 13, 14, 17, 24],
 *   [18, 21, 23, 26, 30]
 * ]
 * 给定 target = 5，返回 true。
 *
 * 给定 target = 20，返回 false。
 * https://leetcode.cn/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/
 *
 * @author lchenglong
 * @date 2022/3/17
 */
public class Offer04 {

    public static void main(String[] args) {
        Offer04 offer04 = new Offer04();
        int[][] array = new int[][]{{1,4,7,11,15},{2,5,8,12,19},{3,6,9,16,22}};
        boolean numberIn2DArray = offer04.findNumberIn2DArray2(array, 22);
        System.out.println(numberIn2DArray);
    }
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        // 外层数组长度
        int m = matrix.length;
        // 内层数组长度
        int n = matrix[0].length;
        int len = Math.min(m,n);
        for (int i = 0;i<len;i++) {
            if (matrix[i][i] == target){
                return true;
            } else if (matrix[i][i]>target) {
                // 往后 往下找[i-1][i-1]
                for (int j = i-1;j>=0;j--){
                    if (matrix[i][j] == target || matrix[j][i] == target){
                        return true;
                    }
                }
            }
        }
        if (m>len){
            for (int j=len-1;j>=0;j--){
                for (int i = len;i<m;i++){
                    if (target == matrix[i][j]){
                        return true;
                    }
                }
            }
        } else {
            for (int j=len-1;j>=0;j--){
                for (int i = len;i<n;i++){
                    if (target == matrix[j][i]){
                        return true;
                    }
                }
            }
        }
        return false;
    }

    public boolean findNumberIn2DArray2(int[][] matrix, int target) {
        // 边界条件
        if (matrix == null || matrix.length == 0|| matrix[0].length == 0){
            return false;
        }
        // 外层数组长度
        int r = matrix.length;
        // 内层数组长度
        int l = matrix[0].length;
        int i = 0,j = l-1;
        while (i<r&&j>=0){
            if (matrix[i][j] == target){
                return true;
            } else if (matrix[i][j]> target){
                //  大于目标 按列往前遍历
                j--;
            } else {
               // 小于目标 按行往后遍历
                i++;
            }
        }
        return false;
    }


}
